// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_RENDERER_PEPPER_PPB_BUFFER_IMPL_H_
#define CONTENT_RENDERER_PEPPER_PPB_BUFFER_IMPL_H_

#include <stdint.h>

#include <memory>

#include "base/macros.h"
#include "base/memory/ref_counted.h"
#include "base/memory/shared_memory.h"
#include "ppapi/shared_impl/resource.h"
#include "ppapi/thunk/ppb_buffer_api.h"

namespace content {

class PPB_Buffer_Impl : public ppapi::Resource,
                        public ppapi::thunk::PPB_Buffer_API {
public:
    static PP_Resource Create(PP_Instance instance, uint32_t size);
    static scoped_refptr<PPB_Buffer_Impl> CreateResource(PP_Instance instance,
        uint32_t size);

    virtual PPB_Buffer_Impl* AsPPB_Buffer_Impl();

    base::SharedMemory* shared_memory() const { return shared_memory_.get(); }
    uint32_t size() const { return size_; }

    // Resource overrides.
    ppapi::thunk::PPB_Buffer_API* AsPPB_Buffer_API() override;

    // PPB_Buffer_API implementation.
    PP_Bool Describe(uint32_t* size_in_bytes) override;
    PP_Bool IsMapped() override;
    void* Map() override;
    void Unmap() override;

    // Trusted.
    int32_t GetSharedMemory(base::SharedMemory** shm) override;

private:
    ~PPB_Buffer_Impl() override;

    explicit PPB_Buffer_Impl(PP_Instance instance);
    bool Init(uint32_t size);

    std::unique_ptr<base::SharedMemory> shared_memory_;
    uint32_t size_;
    int map_count_;

    DISALLOW_COPY_AND_ASSIGN(PPB_Buffer_Impl);
};

// Ensures that the given buffer is mapped, and returns it to its previous
// mapped state in the destructor.
class BufferAutoMapper {
public:
    explicit BufferAutoMapper(ppapi::thunk::PPB_Buffer_API* api);
    ~BufferAutoMapper();

    // Will be NULL on failure to map.
    void* data() { return data_; }
    uint32_t size() { return size_; }

private:
    ppapi::thunk::PPB_Buffer_API* api_;

    bool needs_unmap_;

    void* data_;
    uint32_t size_;

    DISALLOW_COPY_AND_ASSIGN(BufferAutoMapper);
};

} // namespace content

#endif // CONTENT_RENDERER_PEPPER_PPB_BUFFER_IMPL_H_
